package com.joysuch.wwyt.util;

import javafx.util.Pair;
import org.apache.poi.hssf.usermodel.*;

import java.util.List;

/**
 * POI 工具类
 *
 * @Author ZhangDong
 * @Date 2020/7/3 14:10
 */
public class PoiUtil {

    /**
     * @param sheetName sheet名称
     * @param cellNames 列名
     * @return
     */
    public static Pair<HSSFWorkbook, HSSFSheet> getPoi(String sheetName, List<String> cellNames) {
        //声明一个工作簿
        HSSFWorkbook hwb = new HSSFWorkbook();
        //声明一个单子并命名
        HSSFSheet sheet = hwb.createSheet(sheetName);
        //给单子名称一个长度
        sheet.setDefaultColumnWidth((short) 15);
        //生成一个样式
        HSSFCellStyle style = hwb.createCellStyle();
        //创建第一行（也可以成为表头）
        HSSFRow row = sheet.createRow(0);
        //样式字体居中
        //style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        //给表头第一行一次创建单元格
        HSSFCell cell;
        for (int i = 0; i < cellNames.size(); i++) {
            cell = row.createCell((short) i);
            cell.setCellValue(cellNames.get(i));
            cell.setCellStyle(style);
        }
        return new Pair<>(hwb, sheet);
    }


     public static Pair<HSSFWorkbook, HSSFSheet> getExamPoi(String sheetName, List<String> cellNames) {
        //声明一个工作簿
        HSSFWorkbook hwb = new HSSFWorkbook();
        //声明一个单子并命名
        HSSFSheet sheet = hwb.createSheet(sheetName);
        //给单子名称一个长度
        sheet.setDefaultColumnWidth((short) 15);
        //生成一个样式i
        HSSFCellStyle style = hwb.createCellStyle();
        //创建第一行（也可以成为表头）
        HSSFRow row = sheet.createRow(1);
        //样式字体居中
        //style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        //给表头第一行一次创建单元格
        HSSFCell cell;
        for (int i = 0; i < cellNames.size(); i++) {
            cell = row.createCell((short) i);
            cell.setCellValue(cellNames.get(i));
            cell.setCellStyle(style);
        }
        return new Pair<>(hwb, sheet);
    }
}
